import os
import sys


if __name__ == '__main__':

    task = sys.argv[1]
    dest1 = list(map(int,sys.argv[2].split(',')))
    dest2 = list(map(int,sys.argv[3].split(',')))
    dest = zip(dest1,dest2)


    for split_id,samp_num in dest:
        for i in range(1,samp_num+1):
            os.system(f"""python HGT_main.py --task {task}  --model HGT --batch_size 1000 --num_workers 1 --fanout full \
    --relation_method max2min --device_type cuda --feature_type source --initial_sample_method full \
    --n_heads 8 --n_hid 512 --ensemble_num 5  --n_layers 2 --sample_layers 2 \
    --hyperopt bayes --n_epoch 40   --opt_iter 3  \
    --processed_data_path  full_processed_data/  --exp_id {split_id} --exp_train_id {i} \
    --graph_refresh no""")